/**
 * 
 */
/**
 * @author HS
 *
 */
package com.sys.console.realm;

import java.util.Date;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

import javax.annotation.PostConstruct;

import org.apache.log4j.Logger;
import org.apache.shiro.SecurityUtils;
import org.apache.shiro.authc.AuthenticationException;
import org.apache.shiro.authc.AuthenticationInfo;
import org.apache.shiro.authc.AuthenticationToken;
import org.apache.shiro.authc.SimpleAuthenticationInfo;
import org.apache.shiro.authc.UsernamePasswordToken;
import org.apache.shiro.authc.credential.HashedCredentialsMatcher;
import org.apache.shiro.authz.AuthorizationInfo;
import org.apache.shiro.authz.SimpleAuthorizationInfo;
import org.apache.shiro.realm.AuthorizingRealm;
import org.apache.shiro.session.Session;
import org.apache.shiro.subject.PrincipalCollection;
import org.apache.shiro.subject.SimplePrincipalCollection;
import org.apache.shiro.subject.Subject;
import org.springframework.beans.factory.annotation.Autowired;

import com.sys.console.business.SysRoleService;
import com.sys.console.business.SysUserService;
import com.sys.console.cookie.Constant_Cookie;
import com.sys.console.pojo.SysRoleLinkAuthorithDto;
import com.sys.console.pojo.Sys_User;
import com.sys.object.common.utils.DateUtils;
import com.sys.object.common.utils.IpUtils;

/**
 * 自定义的指定Shiro验证用户登录的类
 * 
 * @see 在本例中定义了2个用户:jadyer和玄玉,jadyer具有admin角色和admin:manage权限,玄玉不具有任何角色和权限
 */
public class MyRealm extends AuthorizingRealm {
	public final Logger sysLoger = Logger.getLogger("sysLoger");
	@Autowired
	SysRoleService sysRoleService;
	@Autowired
	SysUserService sysUserService;

	/**
	 * 权限认证
	 */
	@Override
	protected AuthorizationInfo doGetAuthorizationInfo(PrincipalCollection principalCollection) {
		try {
			// 获取登录时输入的用户名
			String loginName = (String) principalCollection.fromRealm(getName()).iterator().next();
			// 到数据库查是否有此对象
			Sys_User user = sysUserService.findByName(loginName);
			if (user != null) {
				sysLoger.info(DateUtils.foramteDate_hh_mm_ss(new Date())+"	"+IpUtils.getIpAddr()+"	doGetAuthorizationInfo	"+user.getUsername()+"	权限验证成功");
				// 权限信息对象info,用来存放查出的用户的所有的角色（role）及权限（permission）
				SimpleAuthorizationInfo info = new SimpleAuthorizationInfo();
				List<SysRoleLinkAuthorithDto> list = sysRoleService.queryRoleAuthorityListByUid(user.getUid());
				//System.out.println(JsonUtil.listToJson(list));
				Set<String> setRole = new HashSet<String>();
				for (SysRoleLinkAuthorithDto role : list) {
					setRole.add(role.getTag());
					info.addStringPermissions(role.getAuthSet());
				}
				info.setRoles(setRole);
				return info;
			}
		} catch (Exception ex) {
			ex.printStackTrace();
		}
		return null;
	}

	/**
	 * 登录认证
	 */
	@Override
	protected AuthenticationInfo doGetAuthenticationInfo(AuthenticationToken authenticationToken)
			throws AuthenticationException {
		try {
			UsernamePasswordToken token = (UsernamePasswordToken) authenticationToken;
			// 查出是否有此用户
			Sys_User user = sysUserService.findByName(token.getUsername());
			if (user != null) {
				sysLoger.info(DateUtils.foramteDate_hh_mm_ss(new Date())+"	doGetAuthenticationInfo	"+user.getUsername()+"	登录验证成功");
				// 若存在，将此用户存放到登录认证info中
				this.setSession(Constant_Cookie.CURRENT_USER_SESSION, user);
				// HttpServletResponse response = ((ServletRequestAttributes)RequestContextHolder.getRequestAttributes()).getResponse();
				//HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
				/***
				 * 如果没有getAndClearSavedRequest,则系统会默认上次访问的地址,
				 */
				//org.apache.shiro.web.util.WebUtils.getAndClearSavedRequest(request);
				//org.apache.shiro.web.util.WebUtils.redirectToSavedRequest(request, response, "/index.do");//重定向地址
				return new SimpleAuthenticationInfo(user.getUsername(), user.getPassword(), getName());
			}
		} catch (Exception ex) {
			ex.printStackTrace();
			return null;
		}
		return null;
	}

	@PostConstruct
	public void initCredentialsMatcher() {// MD5加密
		HashedCredentialsMatcher matcher = new HashedCredentialsMatcher("MD5");
		setCredentialsMatcher(matcher);
	}

	/**
	 * 更新用户授权信息缓存.
	 */
	public void clearCachedAuthorizationInfo(String principal) {
		SimplePrincipalCollection principals = new SimplePrincipalCollection(principal, getName());
		clearCachedAuthorizationInfo(principals);
	}
	/**
	 * 为当前登录的Subject授予角色和权限
	 * 
	 * @see 经测试:本例中该方法的调用时机为需授权资源被访问时
	 * @see 经测试:并且每次访问需授权资源时都会执行该方法中的逻辑,这表明本例中默认并未启用AuthorizationCache
	 * @see 个人感觉若使用了Spring3.1开始提供的ConcurrentMapCache支持,
	 *      则可灵活决定是否启用AuthorizationCache
	 * @see 比如说这里从数据库获取权限信息时,先去访问Spring3.1提供的缓存,而不使用Shior提供的AuthorizationCache
	 */
	/*
	 * @Override protected AuthorizationInfo
	 * doGetAuthorizationInfo(PrincipalCollection principals){
	 * //获取当前登录的用户名,等价于(String)principals.fromRealm(this.getName()).iterator().
	 * next() String currentUsername =
	 * (String)super.getAvailablePrincipal(principals); // List<String> roleList
	 * = new ArrayList<String>(); // List<String> permissionList = new
	 * ArrayList<String>(); // //从数据库中获取当前登录用户的详细信息 // User user =
	 * userService.getByUsername(currentUsername); // if(null != user){ //
	 * //实体类User中包含有用户角色的实体类信息 // if(null!=user.getRoles() &&
	 * user.getRoles().size()>0){ // //获取当前登录用户的角色 // for(Role role :
	 * user.getRoles()){ // roleList.add(role.getName()); //
	 * //实体类Role中包含有角色权限的实体类信息 // if(null!=role.getPermissions() &&
	 * role.getPermissions().size()>0){ // //获取权限 // for(Permission pmss :
	 * role.getPermissions()){ //
	 * if(!StringUtils.isEmpty(pmss.getPermission())){ //
	 * permissionList.add(pmss.getPermission()); // } // } // } // } // } //
	 * }else{ // throw new AuthorizationException(); // } // //为当前用户设置角色和权限 //
	 * SimpleAuthorizationInfo simpleAuthorInfo = new SimpleAuthorizationInfo();
	 * // simpleAuthorInfo.addRoles(roleList); //
	 * simpleAuthorInfo.addStringPermissions(permissionList);
	 * SimpleAuthorizationInfo simpleAuthorInfo = new SimpleAuthorizationInfo();
	 * //实际中可能会像上面注释的那样从数据库取得 if(null!=currentUsername &&
	 * "jadyer".equals(currentUsername)){ //添加一个角色,不是配置意义上的添加,而是证明该用户拥有admin角色
	 * simpleAuthorInfo.addRole("admin"); //添加权限
	 * simpleAuthorInfo.addStringPermission("admin:manage");
	 * System.out.println("已为用户[jadyer]赋予了[admin]角色和[admin:manage]权限"); return
	 * simpleAuthorInfo; }else if(null!=currentUsername &&
	 * "玄玉".equals(currentUsername)){ System.out.println("当前用户[玄玉]无授权"); return
	 * simpleAuthorInfo; } //若该方法什么都不做直接返回null的话,就会导致任何用户访问/admin/listUser.
	 * jsp时都会自动跳转到unauthorizedUrl指定的地址 //详见applicationContext.xml中的<bean
	 * id="shiroFilter">的配置 return null; }
	 * 
	 * 
	 *//**
		 * 验证当前登录的Subject
		 * 
		 * @see 经测试:本例中该方法的调用时机为LoginController.login()方法中执行Subject.login()时
		 *//*
		 * @Override protected AuthenticationInfo
		 * doGetAuthenticationInfo(AuthenticationToken authcToken) throws
		 * AuthenticationException { //获取基于用户名和密码的令牌
		 * //实际上这个authcToken是从LoginController里面currentUser.login(token)传过来的
		 * //两个token的引用都是一样的,本例中是org.apache.shiro.authc.
		 * UsernamePasswordToken@33799a1e UsernamePasswordToken token =
		 * (UsernamePasswordToken)authcToken;
		 * System.out.println("验证当前Subject时获取到token为" +
		 * ReflectionToStringBuilder.toString(token,
		 * ToStringStyle.MULTI_LINE_STYLE)); // User user =
		 * userService.getByUsername(token.getUsername()); // if(null != user){
		 * // AuthenticationInfo authcInfo = new
		 * SimpleAuthenticationInfo(user.getUsername(), user.getPassword(),
		 * user.getNickname()); // this.setSession("currentUser", user); //
		 * return authcInfo; // }else{ // return null; // }
		 * //此处无需比对,比对的逻辑Shiro会做,我们只需返回一个和令牌相关的正确的验证信息
		 * //说白了就是第一个参数填登录用户名,第二个参数填合法的登录密码(可以是从数据库中取到的,本例中为了演示就硬编码了)
		 * //这样一来,在随后的登录页面上就只有这里指定的用户和密码才能通过验证
		 * if("jadyer".equals(token.getUsername())){ AuthenticationInfo
		 * authcInfo = new SimpleAuthenticationInfo("jadyer", "jadyer",
		 * this.getName()); this.setSession("currentUser", "jadyer"); return
		 * authcInfo; }else if("玄玉".equals(token.getUsername())){
		 * AuthenticationInfo authcInfo = new SimpleAuthenticationInfo("玄玉",
		 * "xuanyu", this.getName()); this.setSession("currentUser", "玄玉");
		 * return authcInfo; } //没有返回登录用户名对应的SimpleAuthenticationInfo对象时,
		 * 就会在LoginController中抛出UnknownAccountException异常 return null; }
		 */

	/**
	 * 将一些数据放到ShiroSession中,以便于其它地方使用
	 * 
	 * @see 比如Controller,使用时直接用HttpSession.getAttribute(key)就可以取到
	 */
	private void setSession(Object key, Object value) {
		Subject currentUser = SecurityUtils.getSubject();
		if (null != currentUser) {
			Session session = currentUser.getSession();
			if (null != session) {
				session.setAttribute(key, value);
			}
		}
	}
}